from pymysql import Connection


def query(sql):
    conn = None
    try:
        # 创建数据库连接
        conn = Connection(
            host='localhost',  # 主机地址
            port=3306,  # 端口
            user='root',  # 用户名
            password='root',  # 密码
            database='learn_pymysql',  # 数据库
            autocommit=True  # 配置自动提交
        )
        cursor = conn.cursor()
        cursor.execute(sql)
        return cursor.fetchall()
    except Exception as e:
        print("发生异常：", e)
        return None
    finally:
        if conn:
            # 关闭连接
            conn.close()
